package com.codesynapse.config;

import lombok.Data;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.Configuration;

@Data
@Configuration
@ConfigurationProperties(prefix = "ai.model")
public class AIModelConfig {
    private OllamaConfig ollama;
    private OpenAIConfig openai;

    @Data
    public static class OllamaConfig {
        private boolean enabled;
        private String baseUrl;
        private String modelName;
    }

    @Data
    public static class OpenAIConfig {
        private boolean enabled;
        private String apiKey;
        private String baseUrl;
        private String model;
    }
} 